package com.ruoyi.deal.vo;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.ruoyi.deal.domain.HyDealCategory;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Treeselect树结构实体类
 *
 * @author ruoyi
 */

@Data
@NoArgsConstructor
@Accessors(chain = true)
public class TreeSelect implements Serializable {
    private static final long serialVersionUID = 1L;

    /** 节点ID */
    private Long id;

    /** 节点名称 */
    private String label;

    /** 子节点 */
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    private List<TreeSelect> children;


    public TreeSelect(HyDealCategory category) {
        this.id = category.getCategoryId().longValue();
        this.label = category.getCategoryName();
        this.children = category.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList());
    }
}
